import matplotlib.pyplot as plt
import numpy as np
# 创建一个随机的二维数组作为示例数据
data = np.random.rand(10, 10)
# 使用plt.imshow()显示图像，并指定色彩映射为'viridis'
plt.imshow(data, cmap='viridis')
plt.colorbar()  # 添加颜色条以解释颜色与数据值之间的关系
plt.show()